from django.db import migrations

def insert_keywords(apps, schema_editor):
    Keyword = apps.get_model('trend', 'Keyword')
    
    keywords_data = [
        # 基础理论
        {'id': 1, 'keyword': '人工智能', 'count': 5000},
        {'id': 2, 'keyword': '深度学习', 'count': 4200},
        {'id': 3, 'keyword': '机器学习', 'count': 3800},
        {'id': 4, 'keyword': '神经网络', 'count': 3500},
        {'id': 5, 'keyword': '强化学习', 'count': 2200},
        
        # 计算机视觉
        {'id': 6, 'keyword': '计算机视觉', 'count': 4500},
        {'id': 7, 'keyword': '目标检测', 'count': 3200},
        {'id': 8, 'keyword': '图像识别', 'count': 3000},
        {'id': 9, 'keyword': '人脸识别', 'count': 2800},
        {'id': 10, 'keyword': '视频分析', 'count': 2500},
        
        # 自然语言处理
        {'id': 11, 'keyword': '自然语言处理', 'count': 4000},
        {'id': 12, 'keyword': '机器翻译', 'count': 2800},
        {'id': 13, 'keyword': '语音识别', 'count': 2600},
        {'id': 14, 'keyword': '文本分类', 'count': 2400},
        {'id': 15, 'keyword': '情感分析', 'count': 2200},
    ]
    
    for data in keywords_data:
        Keyword.objects.create(**data)

def reverse_keywords(apps, schema_editor):
    Keyword = apps.get_model('trend', 'Keyword')
    Keyword.objects.all().delete()

class Migration(migrations.Migration):
    dependencies = [
        ('trend', '02_ins_province'),  # 修改为依赖于 02_ins_province
    ]

    operations = [
        migrations.RunPython(insert_keywords, reverse_keywords),
    ]